# 一个文件夹下的图片是数据，另一个文件夹放标签，标签在一个总标签文件夹，所以需要根据图片文件夹的名字从总标签文件夹移动到另一个文件夹
import os
import shutil

# 设置文件夹的路径
image_folder = 'F:\图像处理数据集\VOC2012\images/val'  # 图片文件夹
label_folder = 'F:\图像处理数据集\VOC2012\VOCyololabels\YOLOLabels'  # 总标签文件夹
target_folder = 'F:\图像处理数据集\VOC2012\labels/val'  # 目标文件夹，用于存放移动的标签

# 列出图片文件夹中所有文件的名称（不包括扩展名）
image_files = [os.path.splitext(file)[0] for file in os.listdir(image_folder) if os.path.isfile(os.path.join(image_folder, file))]

# 为每个图片文件查找并移动对应的标签文件
for image_file in image_files:
    # 假设标签文件的扩展名为.txt
    label_file = image_file + '.txt'
    label_path = os.path.join(label_folder, label_file)

    # 检查标签文件是否存在
    if os.path.exists(label_path):
        # 移动标签文件
        shutil.move(label_path, os.path.join(target_folder, label_file))
        print(f"已移动标签文件: {label_file}")
    else:
        print(f"未找到标签文件: {label_file}")

